Theano 简介及其安装

简介

Theano 是一个 Python 科学计算库,允许我们进行符号运算,并在 CPUGPU 上执行。

它最初由 Montreal 大学的机器学习研究者们所开发,用来进行机器学习的计算。

按照官网上的说明,它拥有以下几个方面的特点:

  • Numpy, Scipy 的紧密结合
  • GPU 加速
  • 高效的符号计算
  • 速度和稳定性
  • 动态生成 C 代码

使用 anaconda 安装 theano

windows 下,使用 anaconda 安装 theano 的命令为:

conda install mingw libpython
pip install theano

linux 下,使用 anaconda 安装的命令为

conda install theano

安装好之后,还需要安装 Cuda 并进行 GPU 环境的配置,否则是不能利用 GPU 进行计算的,推荐使用 linux/mac 进行配置,具体方法可以参考官网上的配置说明。

查看安装的版本:

In [1]:
import theano

theano.__version__
Out[1]:
'0.7.0.dev-54186290a97186b9c6b76317e007844529a352f4'

查看当前使用的 device:

In [2]:
theano.config.device
Out[2]:
'cpu'

运行测试:

In [3]:
theano.test()
/usr/local/lib/python2.7/dist-packages/theano/misc/pycuda_init.py:34: UserWarning: PyCUDA import failed in theano.misc.pycuda_init
  warnings.warn("PyCUDA import failed in theano.misc.pycuda_init")
....................S...............
Theano version 0.7.0.dev-54186290a97186b9c6b76317e007844529a352f4
theano is installed in /usr/local/lib/python2.7/dist-packages/theano
NumPy version 1.10.1
NumPy relaxed strides checking option: True
NumPy is installed in /usr/lib/python2.7/dist-packages/numpy
Python version 2.7.6 (default, Jun 22 2015, 17:58:13) [GCC 4.8.2]
nose version 1.3.7
----------------------------------------------------------------------
Ran 37 tests in 37.919s

OK (SKIP=1)
Out[3]:
<nose.result.TextTestResult run=37 errors=0 failures=0>

这里我已经在本地 Windows 配好了 GPU 的设置,如果没有配好,显示的结果可能不一样。

Windows 下第一次运行可能会显示 DEBUG: nvcc STDOUT 等内容,Just ignore it!